#!/usr/bin/env bash
base_dir=$(
  cd $(dirname $0)
  pwd
)
# 获取 cpu 当前使用率
cpu_using=$(top -n 3 | grep '%Cpu' | awk '{print $2}' | awk -F '.' '{print $1}' | tail -1)
# 需要达到的 cpu 使用率，脚本传参
cpu_used=$1

# 如果脚本执行没有参数传入，返回脚本使用方法并退出脚本
if [ $# != 1 ]; then
  printf "\e[0;34mUSAGE: bash $0 30\e[0m\n"
  exit 1
fi

# 如果需要达到的 cpu 使用率小于等于 cpu 当前使用率则退出脚本
## 反之，定义 cpu_status 为 需要达到的 cpu 使用率 和 cpu 当前使用率 的差值
if [[ "${cpu_used}" -le "${cpu_using}" ]]; then
  exit 0
else
  cpu_status=$(echo "$((${cpu_using} - ${cpu_used}))")
fi

# 获取 cpu 线程数
cpu_proc=$(grep 'processor' /proc/cpuinfo | uniq | wc -l)
# cpu 当前使用的线程数量
cpu_using_count=$(awk "BEGIN {printf ${cpu_proc} * ${cpu_using} / 100}" | xargs printf "%.0f")
# 需要达到的 cpu 使用率使用线程数量
cpu_used_count=$(awk "BEGIN {print ${cpu_proc} * ${cpu_used} / 100}" | xargs printf "%.0f")
# 需要增加的线程数量
cpu_num=$(awk "BEGIN {print ${cpu_used_count} - ${cpu_using_count}}" | xargs printf "%.0f")

>${base_dir}/kill_cpu_up.log

for i in $(seq ${cpu_num}); do
  echo -ne "
  i=0;
while true
do
  i=i+1;
done" | /bin/sh &
  echo "kill $!" >>${base_dir}/kill_cpu_up.log 2>&1
done
